package com.spring.housing1.other;

import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.slf4j.Logger;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;


import java.io.*;

@Component
public  class FileUpload {


    // 文件在服务器端保存的主目录
    private static String address;

    private static String host;
    // 端口

    private static int port;
    // ftp用户名

    private static String userName;
    // ftp用户密码

    private static String passWord;
    // 文件在服务器端保存的主目录

    private  static String basePath;

    @Value("${FTP.returnUrl}")
    public void setAddress(String address) {
        this.address = address;
    }

    @Value("${FTP.ADDRESS}")
    public void setHost(String host) {
        this.host = host;
    }
    @Value("${FTP.PORT}")
    public void setPort(int port) {
        this.port = port;
    }
    @Value("${FTP.USERNAME}")
    public void setUserName(String userName) {
        this.userName = userName;
    }
    @Value("${FTP.PASSWORD}")
    public void setPassWord(String passWord) {
        this.passWord = passWord;
    }
    @Value("${FTP.BASEPATH}")
    public void setBasePath(String basePath) {
        this.basePath = basePath;
    }


    /**
    * Description: 向FTP服务器上传文件判断文件服务器中是否有该路径
    * @param path /登录人id/病单id/
    */
    public static String  isPath(String path, Logger log,FTPClient ftp) throws IOException {
String res ="";

                if (!ftp.changeWorkingDirectory(basePath+path)) {
                    return res;
                }else {
                    ftp.enterLocalPassiveMode();//设置被动模式，开通一个端口来传输数据
                    FTPFile[] ftpFiles = ftp.listFiles();
                    int length = ftpFiles.length;
                    for (int i = 0; i <length ; i++) {
                        res += (";"+address + path +ftpFiles[i].getName());
                  }
                    log.info(ftp.printWorkingDirectory());
                    log.info(basePath+path+":{}",length);
                }

        return res;
    }

    /**
     * Description: 向FTP服务器上传文件
     * @param filePath FTP服务器文件存放路径。
     * @param filename 上传到FTP服务器上的文件名
     * @param input 输入流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(
            String filePath, String filename, InputStream input) {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            login(ftp);
            ftp.setFileTransferMode(FTP.STREAM_TRANSFER_MODE);
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //切换到上传目录
            if (!ftp.changeWorkingDirectory(basePath+filePath)) {

                //如果目录不存在创建目录
                String[] dirs = filePath.split("/");
                //linux环境下需要把配置的前置目录填上
                String tempPath = basePath+"";
                for (String dir : dirs) {
                    if (null == dir || "".equals(dir)) continue;
                    tempPath += "/" + dir;
                    if (!ftp.changeWorkingDirectory(tempPath)) {
                        if (!ftp.makeDirectory(tempPath)) {
                            return result;
                        } else {
                            ftp.changeWorkingDirectory(tempPath);
                        }
                    }
                }
            }
            //设置为主动模式(需要服务器tls证书，这里改为主动模式)
            ftp.enterLocalPassiveMode();//设置被动模式，开通一个端口来传输数据
            //设置上传文件的类型为二进制类型
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //上传文件
            if (!ftp.storeFile(filename, input)) {
                return result;
            }
            input.close();
            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }


    //登录
    public static boolean login(FTPClient ftp){

        boolean result = false;
        try {
            int reply;
            FileUpload f = new FileUpload();
            System.out.print(host);
            System.out.print(port);
            ftp.connect(host, port);// 连接FTP服务器
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(userName, passWord);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
        } catch (IOException e) {
            e.printStackTrace();
            return result;
        }

        return true;
    }

    /**
     * 删除文件
     * @param filePath 文件地址如/2022/01/01
     * @param filename 文件名称
     * @return
     */
    public static boolean deleteFile(String filePath, String filename){
        FTPClient ftp = new FTPClient();
        login(ftp);

        boolean flag = false;

        if (ftp != null) {

            try {
                ftp.setFileTransferMode(FTP.STREAM_TRANSFER_MODE);
                ftp.setFileType(FTP.BINARY_FILE_TYPE);
                if (ftp.changeWorkingDirectory(basePath+filePath)) {
                    ftp.dele(filename);
                    flag = true;
                }
            } catch (Exception e) {

                e.printStackTrace();

            } finally {
                if (ftp.isConnected()) {
                    try {
                        ftp.disconnect();
                    } catch (IOException ioe) {
                    }
                }
            }
        }
        return flag;

    }

    /**
     * 下载文件
     * @param filePath 图片路径
     * @param filename 需下载的文件名
     * @param savePath 下载后的文件路径
     * @return 返回是否下载成功 true
     */
    public static Boolean downLoadFile(String filePath, String filename, String savePath){

        FTPClient ftp = new FTPClient();
        login(ftp);


        OutputStream os = null;

        if (ftp != null) {
            try {
                ftp.setFileTransferMode(FTP.STREAM_TRANSFER_MODE);
                ftp.setFileType(FTP.BINARY_FILE_TYPE);
                //判断是否存在该目录
                if (!ftp.changeWorkingDirectory(basePath+filePath)) {
                    System.out.println("/" + basePath+filePath + "该目录不存在");
                    return false;
                }
                ftp.enterLocalPassiveMode();//设置被动模式，开通一个端口来传输数据
                FTPFile[] ftpFiles = ftp.listFiles();
                // 判断该目录下是否有文件
                if (ftpFiles == null || ftpFiles.length == 0) {

                    System.out.println("/" + basePath+filePath + "该目录下无文件");
                    return false;
                }
                for(FTPFile file : ftpFiles){
                    if(filename.equals("") || filename.equalsIgnoreCase(file.getName())){//文件名称为""，下载指定文件
                        if(!file.isDirectory()){//是否文件夹
                            File saveFile = new File(savePath + "/" +file.getName());
                            File directory = new File(savePath );
                            if (!directory.exists()){
                                directory.mkdirs();
                            }
                            os = new FileOutputStream(saveFile);
                            ftp.retrieveFile(file.getName(), os);
                            os.close();
                        }
                    }
                }

                return true;

            } catch (IOException e) {

                System.out.println("下载文件失败" + e.getMessage());

            } finally {

                if(null != os){
                    try {
                        os.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return false;

    }

}
